"""
with语句之上下文管理器

with语句的作用：
    Python提供了with语句的写法，既简单又安全。
    文件操作的时候使用with语句可以自动调用关闭文件操作，即使出现异常也会自动调用关闭文件操作。

with语句为何如此强大？
    with语句之所以这么强大，背后是由上下文管理器作支撑的。
    也就是说f = open()中open函数返回的文件对象就是一个上下文管理器对象。

上下文管理器：
    一个类只要实现了__enter__()和__exit__()，通过该类创建的对象我们就称之为上下文管理器。

    1.__enter__表示上文方法，需要返回一个操作文件对象
    2.__exit__表示下文方法，with语句执行完成会自动执行，即使出现异常也会执行该方法

案例需求：
    自定义上下文管理器类，模拟文件操作

案例步骤：
    1.定义一个File类
    2.实现__enter__()和__exit__()
    3.使用with语句完成文件操作

总结：
    Python提供了with语句用于简化资源释放操作，
    使用with语句操作建立在上下文管理器(实现__enter__和__exit__方法)的基础上。
"""


class File:

    def __init__(self, filename, filemode):
        """

        :param filename: 文件名
        :param filemode: 打开模式
        """
        self.filename = filename
        self.filemode = filemode

    def __enter__(self):
        """上文方法

        :return:
        """
        print("进入上文方法")
        self.file = open(self.filename, self.filemode)
        return self.file  # 返回文件对象

    def __exit__(self, exc_type, exc_val, exc_tb):
        """下文方法

        :param exc_type:
        :param exc_val:
        :param exc_tb:
        :return:
        """
        print("进入下文方法")
        self.file.close()


if __name__ == '__main__':
    with File('a.txt', 'w') as f:
        f.write('123')
